import torch
from backbone.our_resnet import res50

dict_path = '/data1/users/zhengzhiyu/mtp_workplace/documents/RSP/Object Detection/work_dirs/faster/faster_rcnn_orpn_r50_fpn_1x_dota10/latest.pth'
state_dict = torch.load(dict_path)['state_dict']
print(state_dict.keys())

encoder = res50()
class ResNet(torch.nn.Module):
    def __init__(self, 
                args, 
                logger,
                classes: int = 1,
                batch_augments = None):
        super(ResNet, self).__init__()